package br.edu.fatecou.reserva.backing.datamodel;

import java.util.ArrayList;
import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;

import br.edu.fatecou.reserva.dominio.Recurso;

public class RecursoDataModel extends ListDataModel<Recurso> implements
		SelectableDataModel<Recurso> {
	
	public RecursoDataModel() {
		super(new ArrayList<Recurso>());
	}
	
	public RecursoDataModel(List<Recurso> lista) {
		super(lista);
	}

	@SuppressWarnings("unchecked")
	@Override
	public Recurso getRowData(String rowKey) {
		List<Recurso> recursos = (List<Recurso>) getWrappedData();
		
		Long id = Long.valueOf(rowKey);
		for (Recurso recurso : recursos) {
			if (recurso.getCodigo().equals(id))
				return recurso;
		}
		
		return null;
	}

	@Override
	public Object getRowKey(Recurso recurso) {
		return recurso.getCodigo();
	}

}
